# 工程配置
MODEL = 'kimi'        # 生成模型
TOTAL_TASK_NUM = 100  # 生成task的数量
RATE_MODEL = 'kimi'   # 评估模型



# PROMPT配置
SHORTFORM_TO_FULL_TASK_TYPES = {
    "exqa": "extractive question answering",
    "mcqa": "multiple-choice question answering",
    "qg": "question generation",
    "qa": "question answering without choices",
    "ynqa": "yes-no question answering",
    "coref": "coreference resolution",
    "paraphrase": "paraphrase generation",
    "paraphrase_id": "paraphrase identification",
    "sent_comp": "sentence completion",
    "sentiment": "sentiment",
    "summarization": "summarization",
    "text_gen": "text generation",
    "topic_class": "topic classification",
    "wsd": "word sense disambiguation",
    "te": "textual entailment",
    "nli": "natural language inference",
}

NUM_OF_TASK_ONE_REQUEST = 8      # 一次给大模型最多的任务数量
PROMPT_START = "请结合文档内容和你自己的知识以及互联网知识，分别提出"
PROMPT_END = "等不同任务类别的指令任务，指令任务不能显现有文档的存在，即不能出现“根据文档”等短语，之后给出问题回答。格式为：<TAKSTYPE>given type</TASKTYPE><QUESTION>你根据文档内容提出的问题</QUESTIONS><ANSWER>你的回答</ANSWER>:"

example_prompt = '''
请结合文档内容和你自己的知识以及互联网知识，分别提出extractive question answering、extractive question answering、question generation、
question answering without choices、yes-no question answering、coreference resolution、paraphrase generation、paraphrase identification、
sentence completion、sentiment、summarization、text generation、topic classification、word sense disambiguation、
textual entailment、natural language inference等不同任务类别的指令任务，并回答。格式为：<TAKSTYPE>given type</TASKTYPE>。
<QUESTION>你根据文档内容提出的问题</QUESTIONS>。<ANSWER>你的回答</ANSWER>
'''